from game import zz_game
from zz import zz_log, zz_img, zz_code
import uiautomator2 as u2

log = zz_log.of()


# 几何王国

class Game(zz_game.ZZ_Game):
    def __init__(self, d, play_count=1):
        super().__init__(d)
        self.play_count = play_count  # 运行次数

    def get_name(self):
        return "几何王国"

    def game_alert(self):
        self.alert_close_by_file("./game/img/几何王国")

    def handle_main(self):
        self.log_msg("等待进入游戏")
        self.img_exit("./game/img/几何王国/game_main.png", 100)
        self.log_msg("已进入游戏")
        self.d.sleep(2)
        self.game_alert()

        self.log_msg("点击开始游戏")
        self.img_click("./game/img/几何王国/go.png")

        if self.img_exit("./game/img/几何王国/close.png", 3):
            self.img_click_if_exists('存在弹窗，进行关闭', "./game/img/几何王国/close.png", "./game/img/几何王国/play.png")

        for i in range(self.play_count):
            self.log_msg(f"开始第{i + 1}场战斗")
            self.log_msg("点击开始战斗")
            self.img_click("./game/img/几何王国/play.png")
            while True:
                if self.img_exit("./game/img/几何王国/play_again.png", 2):
                    self.log_msg("战斗结束")
                    break
                self.log_msg("战斗中", False)
            self.log_msg(f"点击再次挑战")
            self.img_click("./game/img/几何王国/play_again.png")

        return True
